//
//  ProblemOffer22.swift
//  TestProject
//
//  Created by 毕武侠 on 2021/5/9.
//  Copyright © 2021 zhulong. All rights reserved.
//

import UIKit

/*
 剑指 Offer 22. 链表中倒数第k个节点
 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。

 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

 示例：
     给定一个链表: 1->2->3->4->5, 和 k = 2.

     返回链表 4->5.
 */

@objcMembers class ProblemOffer22: NSObject {
    func solution() {
        let root = ListNode(1)
        root.next = ListNode(2)
        root.next?.next = ListNode(3)
        root.next?.next?.next = ListNode(4)
        root.next?.next?.next?.next = ListNode(5)
        print(getKthFromEnd(root, 2)?.val)
    }
    
    /*
     双指针：经典的倒数第k个节点
     1: 创建2个指针，i和j
     2: 先让i执行k次，这是i和j的间距就是k
     3: 然后同时让i和j向下执行，直到i到达尾部
     */
    func getKthFromEnd(_ head: ListNode?, _ k: Int) -> ListNode? {
        var i = head
        var j = head
        var n = 0
        while i != nil {
            if n < k {
                i = i?.next
                n += 1
                continue
            }
            
            i = i?.next
            j = j?.next
        }
        return j
    }
}
